package com.huluspace.filetransfer.controller;

import jakarta.servlet.http.HttpServletResponse;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

@RestController
@RequestMapping("/files")
public class FileController {
  @Value("${file.upload-dir}")
  private String uploadDir;

  /**
   * 处理文件上传请求
   * @param file 上传的文件对象
   * @return 上传结果信息，包含文件保存路径
   * @throws IOException 文件保存过程中可能抛出的IO异常
   */
  @PostMapping("/upload")
  public String upload(@RequestParam("file") MultipartFile file) throws IOException {
    // 构建文件上传路径
    String filePath = uploadDir + file.getOriginalFilename();
    // 创建目标文件
    File dest = new File(filePath);
    // 保存文件
    file.transferTo(dest);
    // 返回上传成功信息
    return "上传成功：" + filePath;
  }


  /**
   * 文件下载接口
   * @param filename 要下载的文件名
   * @param response HTTP响应对象，用于返回文件内容
   * @throws IOException 当文件读取或写入发生错误时抛出
   */
  @GetMapping("/download")
  public void download(@RequestParam("filename") String filename, HttpServletResponse response) throws IOException {
    // 检查文件是否存在
    File file = new File(uploadDir + filename);
    if (!file.exists()) {
      response.setStatus(HttpServletResponse.SC_NOT_FOUND);
      return;
    }
    // 设置响应头信息，指定文件下载格式
    response.setContentType("application/octet-stream");
    response.setHeader("Content-Disposition", "attachment; filename=" + filename);
    // 将文件内容复制到响应输出流中
    Files.copy(file.toPath(), response.getOutputStream());
    response.getOutputStream().flush();
  }


  /**
   * 列出上传目录中的所有文件
   * @return 文件名列表，如果目录不存在或为空则返回空列表
   */
  @GetMapping("/list")
  public List<String> listFiles() {
    // 获取上传目录文件对象，如果目录不存在则创建目录
    File folder = new File(uploadDir);
    if (!folder.exists()) folder.mkdirs();
    // 获取目录中的文件列表
    String[] files = folder.list();
    // 返回文件列表，如果为空则返回空集合
    return files == null ? Collections.emptyList() : Arrays.asList(files);
  }
}

